package com.tanhua.sso.service;

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.tanhua.sso.mapper.UserMapper;
import com.tanhua.sso.pojo.User;
import io.jsonwebtoken.ExpiredJwtException;
import io.jsonwebtoken.JwsHeader;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.rocketmq.spring.core.RocketMQTemplate;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.messaging.MessagingException;
import org.springframework.stereotype.Service;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@Slf4j
public class UserService {
    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    @Autowired
    private UserMapper userMapper;

    @Value("${jwt.secret}")
    private String secret;

    @Autowired
    private RocketMQTemplate rocketMQTemplate;

    /**
     * 用户登入:用户收到验证码后输入验证码，携带手机号和验证码
     * @param phone 手机号
     * @param code  验证码
     * @return
     */
    public String login(String phone, String code) {
        String redisKey="CHECK_CODE_"+phone;

        //定义是否为新用户，默认否
        boolean isNew=false;

        //校验验证码
        String redisData = redisTemplate.opsForValue().get(redisKey);
        if(!StringUtils.equals(code,redisData)){
            return null;//验证码错误
        }
        //验证码校验完成后需要删除
        redisTemplate.delete(redisKey);

        //查询是否是已经注册
//        Wrapper<User> wrapper= Wrappers.
//                <User>emptyWrapper()
//                .eq("mobile",phone);
        QueryWrapper<User> wrapper=new QueryWrapper<>();
                wrapper.eq("mobile",phone);
        User user = userMapper.selectOne(wrapper);

        //未注册
        if(user==null){
            //需要注册
            user = new User();
            user.setMobile(phone);
            user.setPassword(DigestUtils.md5Hex("123456"));

            //新用户注册
            this.userMapper.insert(user);
            isNew=true;
        }

        //生成token
        Map<String, Object> claims = new HashMap<String, Object>();
//        claims.put("mobile", phone);
        claims.put("id", user.getId());

        // 生成token
        String token = Jwts.builder()
//                .setHeader(header)  //header，可省略
                .setClaims(claims) //payload，存放数据的位置，不能放置敏感数据，如：密码等
                .signWith(SignatureAlgorithm.HS256, secret) //设置加密方法和加密盐
//                .setExpiration(new Date(System.currentTimeMillis() + 12000)) //设置过期时间，3秒后过期
                .setExpiration(new DateTime().plusMinutes(200).toDate()) //设置过期时间，2分钟后过期
                .compact();

        //发送用户登入成功的消息MQ
        try {
            Map<String, Object> msg = new HashMap<>();
            msg.put("id",user.getId());
            msg.put("date",System.currentTimeMillis());
            this.rocketMQTemplate.convertAndSend("tanhua-sso-login",msg);
        } catch (MessagingException e) {
            log.error("消息发送失败",e);
        }

        return token+"|"+isNew;
    }

    public User queryUserByToken(String token) {
        try {
            // 通过token解析数据
            Map<String, Object> body = Jwts.parser()
                    .setSigningKey(secret)
                    .parseClaimsJws(token)
                    .getBody();
//            System.out.println(body); //{mobile=1333333333, id=2, exp=1605513392}
            User user = new User();
            user.setId(Long.valueOf(body.get("id").toString()));

            //返回user对象总的mobile，需要查询数据库获取到mobile数据
            //如果每次都查询数据库，会导致性能问题，需要对手机号进行缓存操作
            //缓存数据要设置过期时间，过期时间要与token保持一致
            //如果用户修改了手机号，要同步修改redis中的数据
            String redisKey="TANHUA_USER_MOBILE_"+user.getId();
            if (this.redisTemplate.hasKey(redisKey)){
                String mobile = this.redisTemplate.opsForValue().get(redisKey);
                user.setMobile(mobile);
            }else {
                //查询数据库
                User u = this.userMapper.selectById(user.getId());
                user.setMobile(u.getMobile());

                //将手机号写入redis中
                //jwt中过期的时间单位为秒
                long timeout= Long.valueOf(body.get("exp").toString())*1000-System.currentTimeMillis();
                this.redisTemplate.opsForValue().set(redisKey,u.getMobile(),timeout, TimeUnit.MILLISECONDS);
            }
            return user;

        } catch (ExpiredJwtException e) {
            log.info("token已经过期!"+token);
            System.out.println("token已经过期！");
        } catch (Exception e) {
            log.error("token不合法！"+token,e);
        }
        return null;
    }
}
